Writing data on a single line using savetxt in Python